package com.jiuli_travel.config;

import com.jiuli_travel.websocket.MessageWebSocketHandler;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;

@Configuration
@EnableWebSocket // 启用WebSocket支持
public class WebSocketConfig implements WebSocketConfigurer {

    /**
     * 注册WebSocket端点和消息处理器。
     * 在这里，我们注册了一个名为"/message-push"的端点，使用自定义的MessageWebSocketHandler处理连接和消息。
     * @param registry WebSocketHandlerRegistry，用于注册WebSocket端点和配置选项
     */
    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(new MessageWebSocketHandler(), "/message-push") // 注册MessageWebSocketHandler处理器
                .setAllowedOrigins("*"); // 允许跨域访问
    }
}